package cn.stylefeng.guns.sys.core.listener;
import cn.stylefeng.guns.sys.modular.system.entity.BtyDywh;
import cn.stylefeng.guns.sys.modular.system.service.BtyDywhService;
import cn.stylefeng.roses.core.util.ToolUtil;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import java.util.ArrayList;
import java.util.List;

/*
 * 监听器批处理数据
 * */
public class DywhExcelListener extends AnalysisEventListener<BtyDywh> {

    private BtyDywhService dywhService;

    public DywhExcelListener(){

    }

    public DywhExcelListener(BtyDywhService dywhService){
        this.dywhService = dywhService;
    }


    /**
     * 批处理阈值2000
     */
    private static final int BATCH_COUNT = 100;
    List<BtyDywh> list = new ArrayList<BtyDywh>();

    @Override
    public void invoke(BtyDywh basic, AnalysisContext analysisContext) {
        if(ToolUtil.isNotEmpty(basic.getYyr())){
            String yyr=basic.getYyr().substring(0,10);
            basic.setYyr(yyr);
        }
        list.add(basic);
        if (list.size() >= BATCH_COUNT) {
            saveData();
            list.clear();
        }
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {
        analysisContext.readWorkbookHolder().setDefaultReturnMap(false);
        saveData();
        // log.info("所有数据解析完成！");

    }
    private void saveData(){
        //调用saveData()方法
        dywhService.insertDyWh(list);
    }
}